package com.jxb.three;

import java.util.HashMap;
import java.util.Map;
import java.util.PriorityQueue;

/**
 * 前 K 个高频元素
 *  给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按任意顺序返回答案。
 *  输入: nums = [1,1,1,2,2,3], k = 2
 *  输出: [1,2]
 * @author jiaobo
 * @date Created in 2024/12/26 23:58
 **/
public class TopKFrequent_347 {

    public int[] topKFrequentWithPQ(int[] nums, int k) {
        Map<Integer,Integer> map = new HashMap<>();
        for (int n:nums) {
            map.put(n,map.getOrDefault(n,0)+1);
        }
        PriorityQueue<int[]> minHeap = new PriorityQueue<>(k, (a, b) -> (a[1] - b[1]));
        for (Map.Entry<Integer,Integer> entry : map.entrySet()) {
            int count = entry.getValue();
            if (minHeap.size() == k) {
                if (minHeap.peek()[1] < count) {
                    minHeap.poll();
                    minHeap.offer(new int[]{entry.getKey(), count});
                }
            }else {
                minHeap.offer(new int[]{entry.getKey(),count});
            }
        }
        int[] ret = new int[k];
        for (int i = 0;i<k;++i) {
            ret[i] = minHeap.poll()[0];
        }
        return ret;
    }

}
